# load data
import pandas as pd
import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

# %matplotlib inline

import warnings
warnings.filterwarnings("ignore")

# Set the display options to show all columns without truncation
pd.set_option('display.max_columns', None)

#Load the data
train = pd.read_csv("widsdatathon2024-challenge1/training.csv")
test = pd.read_csv("widsdatathon2024-challenge1/test.csv")
sample_submission = pd.read_csv("widsdatathon2024-challenge1/sample_submission.csv")

# print(train.shape)
#all columns linked to geo-demographic
train.columns[13:82]
geo_demo_list = train.columns[13:82].tolist()
print(geo_demo_list)